Machine-learning prediction model